ARLoop:带注意力残差的完全开源预训练循环 Transformer
🧭 高效 LLM 预训练 · 循环 Transformer ★ 深度解读
The Ohio State University · Saarland University
提出 ARLoop,用滑动窗口深度注意力改进循环 Transformer,1.2B/0.5T 模型超过 Huginn-3.5B 并接近 Qwen3.5-2B
📌 工作参考价值(4 点)
- 循环 Transformer 是低成本 LLM 基座的可选路线:如果你的业务要用 LLM 做 Agent/生成式推荐,但 GPU 预算有限,ARLoop 这种「共享层 + 多次 recurrence」的架构可以显著降低参数量,同时保留较深的有效计算图;尤其数学/代码类推理任务收益明显。
- 滑动窗口深度注意力是可直接复用的 trick:在循环结构中,每个子层只用 embedding + 最近 3 个 residual block 做深度注意力,比全量历史更省显存/带宽,吞吐损失约 10%,但效果不降反升;这适合你自研深层共享模型时直接套用。
- 固定 recurrence 训练,推理时不要超过训练最大值:论文发现动态 recurrence 不如固定值;推理时 recurrences 超过训练最大值也无法继续涨点。线上 serving 时可以把最大 recurrences 固定到训练值,避免无效计算。
- Value residuals 对深层循环模型稳定性很重要:尤其搭配 Sandwich Norm 时,没有 value residuals 会出现表示坍塌、训练失败;如果你自己训练很深的共享层模型,可以加入 gated token embedding 注入来稳定训练。
循环 MoE 超越两轮的可扩展训练方案
🧭 循环 MoE 深度扩展与训练稳定化 ★ 深度解读
Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences · Peng Cheng Laboratory · University of Chinese Academy of Sciences · The Hong Kong Polytechnic University · University of Surrey
LOOM 用残差缩放、输入重注入、逐循环路由与 Looping Residual 让 MoE LLM 稳定循环 9-12 轮并优于基线
📌 工作参考价值(4 点)
- 在生成式推荐或 Query 生成中若采用 Looped LLM / 共享层 MoE,可复用 γ=λ/(H√M) 的残差缩放和逐轮输入重注入,防止多轮迭代后表示漂移与训练发散,尤其适合反复生成候选 query 或语义 ID。
- 在 FLOPs 受限的在线推理中,建议共享专家但为每轮循环配置独立 router:只增加少量路由参数,就能避免专家选择坍缩,让每轮实际激活不同 expert,提升循环计算收益。
- Looping Residual 以固定衰减 EMA 仅存一个累加张量就能保留跨循环历史,适合在低延迟排序/召回中叠加多轮细化信号,显存开销可忽略。
- 训练深层循环模型时采用分段反向传播(每 K=3 循环为一段、段间 detach),峰值显存几乎不随循环数增长,训练更快更稳,值得 GPU/NPU 受限的业务复现。
门控循环Transformer:通过循环调制实现深度表达力
🧭 高效Transformer架构 · 循环深度共享
The German University in Cairo · Technical University of Munich · Cerebras Systems Inc.
用门控循环共享核心层,在固定算力或参数预算下达到甚至超越稠密Transformer质量,大幅减少参数与解码内存
📌 工作参考价值(4 点)
- 共享核心层 + 固定 prelude/coda 的架构可作为在线大模型推理的轻量化替代:用 3 层共享核心迭代 R 次匹配 12 层稠密模型精度,峰值解码内存降低约 59%,适合内存受限但延迟容忍度较高的生成式推荐/文案生成服务。
- 门控循环更新机制(elementwise gate 结合隐藏状态、prelude 输出与每步重采样噪声)能让同一组层在不同循环步动态调制输入,避免深度共享导致的表达坍缩;该 trick 可迁移到推荐序列建模中的共享 Transformer 层,在参数不变的情况下提升序列多样性建模能力。
- 采用 isoFLOPS / isoPARAMS 评估方式做架构选型,对实际部署有指导意义:在固定算力或参数预算下,优先考虑增加循环深度而非堆叠唯一层,可以获得更好的 loss/质量与内存 tradeoff。
- 对于广告文案生成、搜索 query 推荐等需要大模型但受限于显存和参数规模的场景,GRT 提供了一种可复用的参数高效化路径,且其代码已开源,便于快速验证迁移效果。